package gold.digger;

import java.util.*;
import java.util.List;

/**
 * Created by fanzhenyu02 on 2020/6/27.
 * common problem solver template.
 */
public class LC1130 {
    public long startExecuteTime = System.currentTimeMillis();


    /*
     * @param 此题目参考了别人代码
     * 这是因为问题情况较为复杂
     * 未来需要再次复习此道题目
     * 这题目太难了，难哭了
     * 有两种做法，一种是递归分治，找到一个分解点，拆分成两部分，递归出两部分最大叶节点，得到汇总结果
     * 在递归遍历子部分中，得到递归部分的子解。
     * 另一种就是贪心的让小值沉在二叉树底部，然后用最小栈弹栈累加乘积，这种方式参考：https://leetcode-cn.com/problems/minimum-cost-tree-from-leaf-values/solution/wei-shi-yao-dan-diao-di-jian-zhan-de-suan-fa-ke-xi/
     * 没有严格的理论证明，难哭了
     * @return:
     */
    class Solution {
        public int mctFromLeafValues(int[] arr) {
            Stack<Integer> st = new Stack();
            st.push(Integer.MAX_VALUE);
            int mct = 0;
            for (int i = 0; i < arr.length; i++) {
                while (arr[i] >= st.peek()) {
                    mct += st.pop() * Math.min(st.peek(), arr[i]);
                }
                st.push(arr[i]);
            }
            while (st.size() > 2) {
                mct += st.pop() * st.peek();
            }
            return mct;
        }
    }


    public void run() {
        Solution solution = new Solution();
        List<Integer> list = new ArrayList<>();
        System.out.println(solution.toString());
    }

    public static void main(String[] args) throws Exception {
        LC1130 an = new LC1130();
        an.run();

        System.out.println("\ncurrent solution total execute time: " + (System.currentTimeMillis() - an.startExecuteTime) + " ms.");
    }
}
